from django.db import models
from django.utils.html import format_html


# Create your models here.
class Types(models.Model):
    id = models.AutoField(primary_key=True)
    firsts = models.CharField('一级分类', max_length=100)
    seconds = models.CharField('二级分类', max_length=100)

    def __str__(self):
        return str(self.id)

    class Meta:
        verbose_name = '商品类型'
        verbose_name_plural = '商品类型'


class CommodityInfos(models.Model):
    id = models.AutoField(primary_key=True)
    name = models.CharField('商品名称', max_length=100)
    sezes = models.CharField('颜色规格', max_length=100)
    types = models.CharField('商品类型', max_length=100)
    price = models.FloatField("商品价格")
    discount = models.FloatField("折后价格")
    stock = models.IntegerField("存货数量")
    sold = models.IntegerField("已售数量")
    likes = models.IntegerField("收藏数量")
    created = models.DateTimeField("上架日期", auto_now_add=True)
    img = models.FileField("商品主图", upload_to=r'imgs')
    details = models.FileField("商品介绍", upload_to=r'details')

    def __str__(self):
        return str(self.id)

    class Meta:
        verbose_name = '商品信息'
        verbose_name_plural = '商品信息'

    def colored_name(self):
        if '童装' in self.types:
            color_code = 'red'
        else:
            color_code = 'blue'
        return format_html(
            '<span style="color: {}">{}</span>',
            color_code,
            self.types,
        )

    colored_name.short_description = '带颜色的商品类型'
